package com.gadgets.busi.bm06.dto;

import com.gadgets.busi.bm06.pojo.TradeNodeInfo;

import java.util.ArrayList;
import java.util.List;

/**
 * @author UncleXiu
 * @data 2022-02-27
 * @description
 */
public class TradeNodeTree extends TradeNodeInfo {

    public TradeNodeTree(TradeNodeInfo tradeNodeInfo) {
        super(tradeNodeInfo.getId(),tradeNodeInfo.getNodeName(), tradeNodeInfo.getArea(), tradeNodeInfo.getType());
    }

    private List<TradeNodeTree> treesSon = new ArrayList<>();

    public List<TradeNodeTree> getTreesSon() {
        return treesSon;
    }

    public void setTreesSon(List<TradeNodeTree> treesSon) {
        this.treesSon = treesSon;
    }

    public void addTreeSon(TradeNodeTree treeNode){
        treesSon.add(treeNode);
    }

}
